<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN">

<html lang="en">

<head>
  <meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
  <title>LCOV - code analysis - src/caffe/util/signal_handler.cpp</title>
  <link rel="stylesheet" type="text/css" href="../../../gcov.css">
</head>

<body>

  <table width="100%" border=0 cellspacing=0 cellpadding=0>
    <tr><td class="title">LCOV - code coverage report</td></tr>
    <tr><td class="ruler"><img src="../../../glass.png" width=3 height=3 alt=""></td></tr>

    <tr>
      <td width="100%">
        <table cellpadding=1 border=0 width="100%">
          <tr>
            <td width="10%" class="headerItem">Current view:</td>
            <td width="35%" class="headerValue"><a href="../../../index.html">top level</a> - <a href="index.html">src/caffe/util</a> - signal_handler.cpp<span style="font-size: 80%;"> (source / <a href="signal_handler.cpp.func-sort-c.html">functions</a>)</span></td>
            <td width="5%"></td>
            <td width="15%"></td>
            <td width="10%" class="headerCovTableHead">Hit</td>
            <td width="10%" class="headerCovTableHead">Total</td>
            <td width="15%" class="headerCovTableHead">Coverage</td>
          </tr>
          <tr>
            <td class="headerItem">Test:</td>
            <td class="headerValue">code analysis</td>
            <td></td>
            <td class="headerItem">Lines:</td>
            <td class="headerCovTableEntry">35</td>
            <td class="headerCovTableEntry">49</td>
            <td class="headerCovTableEntryLo">71.4 %</td>
          </tr>
          <tr>
            <td class="headerItem">Date:</td>
            <td class="headerValue">2020-09-11 22:50:33</td>
            <td></td>
            <td class="headerItem">Functions:</td>
            <td class="headerCovTableEntry">7</td>
            <td class="headerCovTableEntry">8</td>
            <td class="headerCovTableEntryMed">87.5 %</td>
          </tr>
          <tr>
            <td class="headerItem">Legend:</td>
            <td class="headerValueLeg">            Lines:
            <span class="coverLegendCov">hit</span>
            <span class="coverLegendNoCov">not hit</span>
</td>
            <td></td>
          </tr>
          <tr><td><img src="../../../glass.png" width=3 height=3 alt=""></td></tr>
        </table>
      </td>
    </tr>

    <tr><td class="ruler"><img src="../../../glass.png" width=3 height=3 alt=""></td></tr>
  </table>

  <table cellpadding=0 cellspacing=0 border=0>
    <tr>
      <td><br></td>
    </tr>
    <tr>
      <td>
<pre class="sourceHeading">          Line data    Source code</pre>
<pre class="source">
<a name="1"><span class="lineNum">       1 </span>            : #include &lt;boost/bind.hpp&gt;</a>
<span class="lineNum">       2 </span>            : #include &lt;glog/logging.h&gt;
<span class="lineNum">       3 </span>            : 
<span class="lineNum">       4 </span>            : #include &lt;signal.h&gt;
<span class="lineNum">       5 </span>            : #include &lt;csignal&gt;
<span class="lineNum">       6 </span>            : 
<span class="lineNum">       7 </span>            : #include &quot;caffe/util/signal_handler.h&quot;
<span class="lineNum">       8 </span>            : 
<span class="lineNum">       9 </span>            : namespace {
<span class="lineNum">      10 </span>            :   static volatile sig_atomic_t got_sigint = false;
<span class="lineNum">      11 </span>            :   static volatile sig_atomic_t got_sighup = false;
<a name="12"><span class="lineNum">      12 </span>            :   static bool already_hooked_up = false;</a>
<span class="lineNum">      13 </span>            : 
<span class="lineNum">      14 </span><span class="lineNoCov">          0 :   void handle_signal(int signal) {</span>
<span class="lineNum">      15 </span><span class="lineNoCov">          0 :     switch (signal) {</span>
<span class="lineNum">      16 </span>            :     case SIGHUP:
<span class="lineNum">      17 </span><span class="lineNoCov">          0 :       got_sighup = true;</span>
<span class="lineNum">      18 </span><span class="lineNoCov">          0 :       break;</span>
<span class="lineNum">      19 </span>            :     case SIGINT:
<span class="lineNum">      20 </span><span class="lineNoCov">          0 :       got_sigint = true;</span>
<span class="lineNum">      21 </span><span class="lineNoCov">          0 :       break;</span>
<span class="lineNum">      22 </span>            :     }
<a name="23"><span class="lineNum">      23 </span><span class="lineNoCov">          0 :   }</span></a>
<span class="lineNum">      24 </span>            : 
<span class="lineNum">      25 </span><span class="lineCov">          1 :   void HookupHandler() {</span>
<span class="lineNum">      26 </span><span class="lineCov">          1 :     if (already_hooked_up) {</span>
<span class="lineNum">      27 </span><span class="lineNoCov">          0 :       LOG(FATAL) &lt;&lt; &quot;Tried to hookup signal handlers more than once.&quot;;</span>
<span class="lineNum">      28 </span>            :     }
<span class="lineNum">      29 </span><span class="lineCov">          1 :     already_hooked_up = true;</span>
<span class="lineNum">      30 </span>            : 
<span class="lineNum">      31 </span>            :     struct sigaction sa;
<span class="lineNum">      32 </span>            :     // Setup the handler
<span class="lineNum">      33 </span><span class="lineCov">          1 :     sa.sa_handler = &amp;handle_signal;</span>
<span class="lineNum">      34 </span>            :     // Restart the system call, if at all possible
<span class="lineNum">      35 </span><span class="lineCov">          1 :     sa.sa_flags = SA_RESTART;</span>
<span class="lineNum">      36 </span>            :     // Block every signal during the handler
<span class="lineNum">      37 </span><span class="lineCov">          1 :     sigfillset(&amp;sa.sa_mask);</span>
<span class="lineNum">      38 </span>            :     // Intercept SIGHUP and SIGINT
<span class="lineNum">      39 </span><span class="lineCov">          1 :     if (sigaction(SIGHUP, &amp;sa, NULL) == -1) {</span>
<span class="lineNum">      40 </span><span class="lineNoCov">          0 :       LOG(FATAL) &lt;&lt; &quot;Cannot install SIGHUP handler.&quot;;</span>
<span class="lineNum">      41 </span>            :     }
<span class="lineNum">      42 </span><span class="lineCov">          1 :     if (sigaction(SIGINT, &amp;sa, NULL) == -1) {</span>
<span class="lineNum">      43 </span><span class="lineNoCov">          0 :       LOG(FATAL) &lt;&lt; &quot;Cannot install SIGINT handler.&quot;;</span>
<span class="lineNum">      44 </span>            :     }
<span class="lineNum">      45 </span><span class="lineCov">          1 :   }</span>
<a name="46"><span class="lineNum">      46 </span>            : </a>
<span class="lineNum">      47 </span>            :   // Set the signal handlers to the default.
<span class="lineNum">      48 </span><span class="lineCov">          1 :   void UnhookHandler() {</span>
<span class="lineNum">      49 </span><span class="lineCov">          1 :     if (already_hooked_up) {</span>
<span class="lineNum">      50 </span>            :       struct sigaction sa;
<span class="lineNum">      51 </span>            :       // Setup the sighup handler
<span class="lineNum">      52 </span><span class="lineCov">          1 :       sa.sa_handler = SIG_DFL;</span>
<span class="lineNum">      53 </span>            :       // Restart the system call, if at all possible
<span class="lineNum">      54 </span><span class="lineCov">          1 :       sa.sa_flags = SA_RESTART;</span>
<span class="lineNum">      55 </span>            :       // Block every signal during the handler
<span class="lineNum">      56 </span><span class="lineCov">          1 :       sigfillset(&amp;sa.sa_mask);</span>
<span class="lineNum">      57 </span>            :       // Intercept SIGHUP and SIGINT
<span class="lineNum">      58 </span><span class="lineCov">          1 :       if (sigaction(SIGHUP, &amp;sa, NULL) == -1) {</span>
<span class="lineNum">      59 </span><span class="lineNoCov">          0 :         LOG(FATAL) &lt;&lt; &quot;Cannot uninstall SIGHUP handler.&quot;;</span>
<span class="lineNum">      60 </span>            :       }
<span class="lineNum">      61 </span><span class="lineCov">          1 :       if (sigaction(SIGINT, &amp;sa, NULL) == -1) {</span>
<span class="lineNum">      62 </span><span class="lineNoCov">          0 :         LOG(FATAL) &lt;&lt; &quot;Cannot uninstall SIGINT handler.&quot;;</span>
<span class="lineNum">      63 </span>            :       }
<span class="lineNum">      64 </span>            : 
<span class="lineNum">      65 </span><span class="lineCov">          1 :       already_hooked_up = false;</span>
<span class="lineNum">      66 </span>            :     }
<span class="lineNum">      67 </span><span class="lineCov">          1 :   }</span>
<span class="lineNum">      68 </span>            : 
<span class="lineNum">      69 </span>            :   // Return true iff a SIGINT has been received since the last time this
<span class="lineNum">      70 </span>            :   // function was called.
<span class="lineNum">      71 </span>            :   bool GotSIGINT() {
<span class="lineNum">      72 </span><span class="lineCov">      12100 :     bool result = got_sigint;</span>
<span class="lineNum">      73 </span><span class="lineCov">      12100 :     got_sigint = false;</span>
<span class="lineNum">      74 </span>            :     return result;
<span class="lineNum">      75 </span>            :   }
<span class="lineNum">      76 </span>            : 
<span class="lineNum">      77 </span>            :   // Return true iff a SIGHUP has been received since the last time this
<span class="lineNum">      78 </span>            :   // function was called.
<span class="lineNum">      79 </span>            :   bool GotSIGHUP() {
<span class="lineNum">      80 </span><span class="lineCov">      12100 :     bool result = got_sighup;</span>
<span class="lineNum">      81 </span><span class="lineCov">      12100 :     got_sighup = false;</span>
<span class="lineNum">      82 </span>            :     return result;
<span class="lineNum">      83 </span>            :   }
<span class="lineNum">      84 </span>            : }  // namespace
<span class="lineNum">      85 </span>            : 
<a name="86"><span class="lineNum">      86 </span>            : namespace caffe {</a>
<span class="lineNum">      87 </span>            : 
<span class="lineNum">      88 </span><span class="lineCov">          1 : SignalHandler::SignalHandler(SolverAction::Enum SIGINT_action,</span>
<span class="lineNum">      89 </span>            :                              SolverAction::Enum SIGHUP_action):
<span class="lineNum">      90 </span>            :   SIGINT_action_(SIGINT_action),
<span class="lineNum">      91 </span><span class="lineCov">          1 :   SIGHUP_action_(SIGHUP_action) {</span>
<span class="lineNum">      92 </span><span class="lineCov">          1 :   HookupHandler();</span>
<a name="93"><span class="lineNum">      93 </span><span class="lineCov">          1 : }</span></a>
<span class="lineNum">      94 </span>            : 
<span class="lineNum">      95 </span><span class="lineCov">          2 : SignalHandler::~SignalHandler() {</span>
<span class="lineNum">      96 </span><span class="lineCov">          1 :   UnhookHandler();</span>
<a name="97"><span class="lineNum">      97 </span><span class="lineCov">          1 : }</span></a>
<span class="lineNum">      98 </span>            : 
<span class="lineNum">      99 </span><span class="lineCov">      12100 : SolverAction::Enum SignalHandler::CheckForSignals() const {</span>
<span class="lineNum">     100 </span><span class="lineCov">      12100 :   if (GotSIGHUP()) {</span>
<span class="lineNum">     101 </span><span class="lineNoCov">          0 :     return SIGHUP_action_;</span>
<span class="lineNum">     102 </span>            :   }
<span class="lineNum">     103 </span><span class="lineCov">      12100 :   if (GotSIGINT()) {</span>
<span class="lineNum">     104 </span><span class="lineNoCov">          0 :     return SIGINT_action_;</span>
<span class="lineNum">     105 </span>            :   }
<span class="lineNum">     106 </span>            :   return SolverAction::NONE;
<span class="lineNum">     107 </span>            : }
<span class="lineNum">     108 </span>            : 
<a name="109"><span class="lineNum">     109 </span>            : // Return the function that the solver can use to find out if a snapshot or</a>
<span class="lineNum">     110 </span>            : // early exit is being requested.
<span class="lineNum">     111 </span><span class="lineCov">          1 : ActionCallback SignalHandler::GetActionFunction() {</span>
<span class="lineNum">     112 </span><span class="lineCov">          1 :   return boost::bind(&amp;SignalHandler::CheckForSignals, this);</span>
<a name="113"><span class="lineNum">     113 </span>            : }</a>
<span class="lineNum">     114 </span>            : 
<span class="lineNum">     115 </span><span class="lineCov">          2 : }  // namespace caffe</span>
</pre>
      </td>
    </tr>
  </table>
  <br>

  <table width="100%" border=0 cellspacing=0 cellpadding=0>
    <tr><td class="ruler"><img src="../../../glass.png" width=3 height=3 alt=""></td></tr>
    <tr><td class="versionInfo">Generated by: <a href="http://ltp.sourceforge.net/coverage/lcov.php" target="_parent">LCOV version 1.12</a></td></tr>
  </table>
  <br>

</body>
</html>
